BZOJ3996【TJOI2015】线性代数 <最大权闭合子图>

Problem

【TJOI2015】线性代数

Description

给出一个的矩阵和一个的矩阵。
求出一个的矩阵，使得最大，输出最大的值。

Input

第一行输入一个整数。
接下来行输入矩阵，第行第个数字代表.
接下来一行输入个整数，代表矩阵。
矩阵和矩阵中每个数字都是不超过的非负整数。

Output

输出最大的。

Sample Input

Sample Output

Hint

标签：网络流 最大权闭合子图

Solution

先化简一下，用和式表示：

由于是矩阵，可以将其看作有个物品，选择取或不去，第给物品取有的代价，第个和第个同时取会有的贡献。要求使总贡献最大。

对于每个物体建一个点，任意两个物体的组合建一个点。对于组合，连接其代表的点和第个物体的点与第个物体的点。对于物体的点，权值设为，对于组合的点，权值设为。这样选择一个组合的点，必须将其连向的两个物体的点也选择，即变为最大权闭合子图模型。

建模：

对每个物体建一个点，连接流量
对每个组合建一个点，连接流量
对于每个组合，连边流量，连边流量

跑最大流求最小割，答案为。

Code

#include <bits/stdc++.h>
#define MAX_N 300000
#define MAX_M 2000000
#define INF 0x3f3f3f3f
using namespace std;
template <class T> inline void read(T &x) {
    x = 0; int c = getchar(), f = 1;
    for (; !isdigit(c); c = getchar()) if (c == 45) f = -1;
    for (; isdigit(c); c = getchar()) (x *= 10) += f*(c-'0');
}
int n, s, t, cnt, tot;
int d[MAX_N+5], pr[MAX_N+5], cr[MAX_N+5];
struct node {int v, c, nxt;} E[MAX_M+5];
void init() {s = 0, t = n*n+n+1, cnt = 0, memset(pr, -1, sizeof pr);}
void insert(int u, int v, int c) {E[cnt] = (node){v, c, pr[u]}, pr[u] = cnt++;}
void addedge(int u, int v, int c) {insert(u, v, c), insert(v, u, 0);}
bool BFS() {
    queue <int> que; que.push(s);
    memset(d, -1, sizeof d), d[s] = 0;
    while (!que.empty()) {
        int u = que.front(); que.pop();
        for (int i = pr[u]; ~i; i = E[i].nxt) {
            int v = E[i].v, c = E[i].c;
            if (~d[v] || !c) continue;
            d[v] = d[u]+1, que.push(v);
        }
    }
    return ~d[t];
}
int DFS(int u, int flow) {
    if (u == t) return flow; int ret = 0;
    for (int &i = pr[u]; ~i; i = E[i].nxt) {
        int v = E[i].v, c = E[i].c;
        if (d[u]+1 != d[v] || !c) continue;
        int tmp = DFS(v, min(flow, c));
        E[i].c -= tmp, E[i^1].c += tmp;
        flow -= tmp, ret += tmp;
        if (!flow) break;
    }
    if (!ret) d[u] = -1;	return ret;
}
void cpy() {for (int i = s; i <= t; i++) cr[i] = pr[i];}
void rec() {for (int i = s; i <= t; i++) pr[i] = cr[i];}
int Dinic() {int ret = 0; cpy(); while (BFS()) ret += DFS(s, INF), rec(); return ret;}
int main() {
    read(n), init();
    for (int i = 1; i <= n; i++)
        for (int j = 1, b; j <= n; j++)
            read(b), tot += b, addedge(s, (i-1)*n+j, b);
    for (int i = 1, c; i <= n; i++)
        read(c), addedge(n*n+i, t, c);
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= n; j++)
            addedge((i-1)*n+j, n*n+i, INF), 
            addedge((i-1)*n+j, n*n+j, INF);
    return printf("%d\n", tot-Dinic()), 0;
}